package Test.tools;

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.Date;


public class DatabaseBackup {

	public static void main(String[] args) {
		System.out.println(new Date()+"开始备份...");
		backup("C://Program Files (x86)//MySQL//MySQL Server 5.0//bin//","127.0.0.1","root","root","bwkj","c:/test.sql");
		System.out.println(new Date()+"备份成功...");

		System.out.println("开始还原...");
		load("C://Program Files (x86)//MySQL//MySQL Server 5.0//bin//","root","root","bwkj","c:/test.sql");
		System.out.println("还原成功...");
	}
	
	
	public static void runCopyDatabase(){
		Date dd=new Date();
		Long ss=dd.getTime();
		String newname=ss+"";
		long start = System.currentTimeMillis();
		System.out.println(new Date()+"开始备份数据库...");
		backup("C://Program Files (x86)//MySQL//MySQL Server 5.0//bin//","127.0.0.1","root","root","bwkj","c:/test_"+newname+".sql");
		long end = System.currentTimeMillis();
		System.out.println(new Date()+"备份成功数据库...花费时间(ms): " + (end - start)+"(ms)");
	}

	public static void backup(String binpath,String databaseip,String username,String password,String database,String outputlocation) {
		try {
			Runtime rt = Runtime.getRuntime();
			// 调用 调用mysql的安装目录的命令
			Process child = rt.exec(binpath+"mysqldump -h "+databaseip+" -u"+username+" -p"+password+"  "+database+"");
			// 设置导出编码为utf-8。这里必须是utf-8
			// 把进程执行中的控制台输出信息写入.sql文件，即生成了备份文件。注：如果不对控制台信息进行读出，则会导致进程堵塞无法运行
			InputStream in = child.getInputStream();// 控制台的输出信息作为输入流
			InputStreamReader xx = new InputStreamReader(in, "utf-8");
			// 设置输出流编码为utf-8。这里必须是utf-8，否则从流中读入的是乱码
			String inStr;
			StringBuffer sb = new StringBuffer("");
			String outStr;
			// 组合控制台输出信息字符串
			BufferedReader br = new BufferedReader(xx);
			while ((inStr = br.readLine()) != null) {
				sb.append(inStr + "\r\n");
			}
			outStr = sb.toString();
			// 要用来做导入用的sql目标文件：
			FileOutputStream fout = new FileOutputStream(outputlocation);
			OutputStreamWriter writer = new OutputStreamWriter(fout, "utf-8");
			writer.write(outStr);
			writer.flush();
			in.close();
			xx.close();
			br.close();
			writer.close();
			fout.close();
			
		} catch (Exception e) {
			e.printStackTrace();
		}
	}


	public static void load(String binpath,String username,String password,String database,String sqllocation) {
		try {
			String fPath = sqllocation;
			Runtime rt = Runtime.getRuntime();

			// 调用 mysql 的 cmd:
			Process child = rt.exec(binpath+"mysql.exe -u"+username+" -p"+password+" "+database+" ");
			OutputStream out = child.getOutputStream();// 控制台的输入信息作为输出流
			String inStr;
			StringBuffer sb = new StringBuffer("");
			String outStr;
			BufferedReader br = new BufferedReader(new InputStreamReader(
					new FileInputStream(fPath), "utf8"));
			while ((inStr = br.readLine()) != null) {
				sb.append(inStr + "\r\n");
			}
			outStr = sb.toString();

			OutputStreamWriter writer = new OutputStreamWriter(out, "utf8");
			writer.write(outStr);
			// 注：这里如果用缓冲方式写入文件的话，会导致中文乱码，用flush()方法则可以避免
			writer.flush();
			// 别忘记关闭输入输出流
			out.close();
			br.close();
			writer.close();

		} catch (Exception e) {
			e.printStackTrace();
		}

	}

}